home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / imputil.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  15KB  |  541 lines

  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. '''
  5. Import utilities
  6.  
  7. Exported classes:
  8.     ImportManager   Manage the import process
  9.  
  10.     Importer        Base class for replacing standard import functions
  11.     BuiltinImporter Emulate the import mechanism for builtin and frozen modules
  12.  
  13.     DynLoadSuffixImporter
  14. '''
  15. import imp
  16. import sys
  17. import __builtin__
  18. import struct
  19. import marshal
  20. __all__ = [
  21.     'ImportManager',
  22.     'Importer',
  23.     'BuiltinImporter']
  24. _StringType = type('')
  25. _ModuleType = type(sys)
  26.  
  27. class ImportManager:
  28.     '''Manage the import process.'''
  29.     
  30.     def install(self, namespace = vars(__builtin__)):
  31.         '''Install this ImportManager into the specified namespace.'''
  32.         if isinstance(namespace, _ModuleType):
  33.             namespace = vars(namespace)
  34.         
  35.         self.previous_importer = namespace['__import__']
  36.         self.namespace = namespace
  37.         namespace['__import__'] = self._import_hook
  38.  
  39.     
  40.     def uninstall(self):
  41.         '''Restore the previous import mechanism.'''
  42.         self.namespace['__import__'] = self.previous_importer
  43.  
  44.     
  45.     def add_suffix(self, suffix, importFunc):
  46.         if not callable(importFunc):
  47.             raise AssertionError
  48.         self.fs_imp.add_suffix(suffix, importFunc)
  49.  
  50.     clsFilesystemImporter = None
  51.     
  52.     def __init__(self, fs_imp = None):
  53.         if not _os_stat:
  54.             _os_bootstrap()
  55.         
  56.         if fs_imp is None:
  57.             if not self.clsFilesystemImporter:
  58.                 pass
  59.             cls = _FilesystemImporter
  60.             fs_imp = cls()
  61.         
  62.         self.fs_imp = fs_imp
  63.         for desc in imp.get_suffixes():
  64.             if desc[2] == imp.C_EXTENSION:
  65.                 self.add_suffix(desc[0], DynLoadSuffixImporter(desc).import_file)
  66.                 continue
  67.         
  68.         self.add_suffix('.py', py_suffix_importer)
  69.  
  70.     
  71.     def _import_hook(self, fqname, globals = None, locals = None, fromlist = None):
  72.         '''Python calls this hook to locate and import a module.'''
  73.         parts = fqname.split('.')
  74.         parent = self._determine_import_context(globals)
  75.         if parent:
  76.             module = parent.__importer__._do_import(parent, parts, fromlist)
  77.             if module:
  78.                 return module
  79.             
  80.         
  81.         
  82.         try:
  83.             top_module = sys.modules[parts[0]]
  84.         except KeyError:
  85.             top_module = self._import_top_module(parts[0])
  86.             if not top_module:
  87.                 raise ImportError, 'No module named ' + fqname
  88.             
  89.         except:
  90.             top_module
  91.  
  92.         if len(parts) == 1:
  93.             if not fromlist:
  94.                 return top_module
  95.             
  96.             if not top_module.__dict__.get('__ispkg__'):
  97.                 return top_module
  98.             
  99.         
  100.         importer = top_module.__dict__.get('__importer__')
  101.         if importer:
  102.             return importer._finish_import(top_module, parts[1:], fromlist)
  103.         
  104.         if len(parts) == 2 and hasattr(top_module, parts[1]):
  105.             return top_module
  106.         
  107.         raise ImportError, 'No module named ' + fqname
  108.  
  109.     
  110.     def _determine_import_context(self, globals):
  111.         '''Returns the context in which a module should be imported.
  112.  
  113.         The context could be a loaded (package) module and the imported module
  114.         will be looked for within that package. The context could also be None,
  115.         meaning there is no context -- the module should be looked for as a
  116.         "top-level" module.
  117.         '''
  118.         if not globals or not globals.get('__importer__'):
  119.             return None
  120.         
  121.         parent_fqname = globals['__name__']
  122.         if globals['__ispkg__']:
  123.             parent = sys.modules[parent_fqname]
  124.             if not globals is parent.__dict__:
  125.                 raise AssertionError
  126.             return parent
  127.         
  128.         i = parent_fqname.rfind('.')
  129.         if i == -1:
  130.             return None
  131.         
  132.         parent_fqname = parent_fqname[:i]
  133.         parent = sys.modules[parent_fqname]
  134.         if not parent.__name__ == parent_fqname:
  135.             raise AssertionError
  136.         return parent
  137.  
  138.     
  139.     def _import_top_module(self, name):
  140.         for item in sys.path:
  141.             if isinstance(item, _StringType):
  142.                 module = self.fs_imp.import_from_dir(item, name)
  143.             else:
  144.                 module = item.import_top(name)
  145.             if module:
  146.                 return module
  147.                 continue
  148.         
  149.  
  150.     
  151.     def _reload_hook(self, module):
  152.         '''Python calls this hook to reload a module.'''
  153.         importer = module.__dict__.get('__importer__')
  154.         if not importer:
  155.             pass
  156.         
  157.         raise SystemError, 'reload not yet implemented'
  158.  
  159.  
  160.  
  161. class Importer:
  162.     '''Base class for replacing standard import functions.'''
  163.     
  164.     def import_top(self, name):
  165.         '''Import a top-level module.'''
  166.         return self._import_one(None, name, name)
  167.  
  168.     
  169.     def _finish_import(self, top, parts, fromlist):
  170.         bottom = self._load_tail(top, parts)
  171.         if not fromlist:
  172.             return top
  173.         
  174.         if bottom.__ispkg__:
  175.             self._import_fromlist(bottom, fromlist)
  176.         
  177.         return bottom
  178.  
  179.     
  180.     def _import_one(self, parent, modname, fqname):
  181.         '''Import a single module.'''
  182.         
  183.         try:
  184.             return sys.modules[fqname]
  185.         except KeyError:
  186.             pass
  187.  
  188.         result = self.get_code(parent, modname, fqname)
  189.         if result is None:
  190.             return None
  191.         
  192.         module = self._process_result(result, fqname)
  193.         if parent:
  194.             setattr(parent, modname, module)
  195.         
  196.         return module
  197.  
  198.     
  199.     def _process_result(self, .2, fqname):
  200.         (ispkg, code, values) = .2
  201.         is_module = isinstance(code, _ModuleType)
  202.         if is_module:
  203.             module = code
  204.         else:
  205.             module = imp.new_module(fqname)
  206.         module.__importer__ = self
  207.         module.__ispkg__ = ispkg
  208.         module.__dict__.update(values)
  209.         sys.modules[fqname] = module
  210.         if not is_module:
  211.             
  212.             try:
  213.                 exec code in module.__dict__
  214.             if fqname in sys.modules:
  215.                 del sys.modules[fqname]
  216.             
  217.  
  218.             raise 
  219.         
  220.         is_module
  221.         module = sys.modules[fqname]
  222.         module.__name__ = fqname
  223.         return module
  224.  
  225.     
  226.     def _load_tail(self, m, parts):
  227.         '''Import the rest of the modules, down from the top-level module.
  228.  
  229.         Returns the last module in the dotted list of modules.
  230.         '''
  231.         for part in parts:
  232.             fqname = '%s.%s' % (m.__name__, part)
  233.             m = self._import_one(m, part, fqname)
  234.             if not m:
  235.                 raise ImportError, 'No module named ' + fqname
  236.                 continue
  237.         
  238.         return m
  239.  
  240.     
  241.     def _import_fromlist(self, package, fromlist):
  242.         '''Import any sub-modules in the "from" list.'''
  243.         if '*' in fromlist:
  244.             fromlist = list(fromlist) + list(package.__dict__.get('__all__', []))
  245.         
  246.         for sub in fromlist:
  247.             if sub != '*' and not hasattr(package, sub):
  248.                 subname = '%s.%s' % (package.__name__, sub)
  249.                 submod = self._import_one(package, sub, subname)
  250.                 if not submod:
  251.                     raise ImportError, 'cannot import name ' + subname
  252.                 
  253.             submod
  254.         
  255.  
  256.     
  257.     def _do_import(self, parent, parts, fromlist):
  258.         '''Attempt to import the module relative to parent.
  259.  
  260.         This method is used when the import context specifies that <self>
  261.         imported the parent module.
  262.         '''
  263.         top_name = parts[0]
  264.         top_fqname = parent.__name__ + '.' + top_name
  265.         top_module = self._import_one(parent, top_name, top_fqname)
  266.         if not top_module:
  267.             return None
  268.         
  269.         return self._finish_import(top_module, parts[1:], fromlist)
  270.  
  271.     
  272.     def get_code(self, parent, modname, fqname):
  273.         '''Find and retrieve the code for the given module.
  274.  
  275.         parent specifies a parent module to define a context for importing. It
  276.         may be None, indicating no particular context for the search.
  277.  
  278.         modname specifies a single module (not dotted) within the parent.
  279.  
  280.         fqname specifies the fully-qualified module name. This is a
  281.         (potentially) dotted name from the "root" of the module namespace
  282.         down to the modname.
  283.         If there is no parent, then modname==fqname.
  284.  
  285.         This method should return None, or a 3-tuple.
  286.  
  287.         * If the module was not found, then None should be returned.
  288.  
  289.         * The first item of the 2- or 3-tuple should be the integer 0 or 1,
  290.             specifying whether the module that was found is a package or not.
  291.  
  292.         * The second item is the code object for the module (it will be
  293.             executed within the new module\'s namespace). This item can also
  294.             be a fully-loaded module object (e.g. loaded from a shared lib).
  295.  
  296.         * The third item is a dictionary of name/value pairs that will be
  297.             inserted into new module before the code object is executed. This
  298.             is provided in case the module\'s code expects certain values (such
  299.             as where the module was found). When the second item is a module
  300.             object, then these names/values will be inserted *after* the module
  301.             has been loaded/initialized.
  302.         '''
  303.         raise RuntimeError, 'get_code not implemented'
  304.  
  305.  
  306. if not __debug__ or 'c':
  307.     pass
  308. _suffix_char = 'o'
  309. _suffix = '.py' + _suffix_char
  310.  
  311. def _compile(pathname, timestamp):
  312.     """Compile (and cache) a Python source file.
  313.  
  314.     The file specified by <pathname> is compiled to a code object and
  315.     returned.
  316.  
  317.     Presuming the appropriate privileges exist, the bytecodes will be
  318.     saved back to the filesystem for future imports. The source file's
  319.     modification timestamp must be provided as a Long value.
  320.     """
  321.     codestring = open(pathname, 'rU').read()
  322.     if codestring and codestring[-1] != '\n':
  323.         codestring = codestring + '\n'
  324.     
  325.     code = __builtin__.compile(codestring, pathname, 'exec')
  326.     
  327.     try:
  328.         f = open(pathname + _suffix_char, 'wb')
  329.     except IOError:
  330.         pass
  331.  
  332.     f.write('\x00\x00\x00\x00')
  333.     f.write(struct.pack('<I', timestamp))
  334.     marshal.dump(code, f)
  335.     f.flush()
  336.     f.seek(0, 0)
  337.     f.write(imp.get_magic())
  338.     f.close()
  339.     return code
  340.  
  341. _os_stat = None
  342. _os_path_join = None
  343.  
  344. def _os_bootstrap():
  345.     """Set up 'os' module replacement functions for use during import bootstrap."""
  346.     global _os_stat, _os_path_join
  347.     names = sys.builtin_module_names
  348.     join = None
  349.     if 'posix' in names:
  350.         sep = '/'
  351.         stat = stat
  352.         import posix
  353.     elif 'nt' in names:
  354.         sep = '\\'
  355.         stat = stat
  356.         import nt
  357.     elif 'dos' in names:
  358.         sep = '\\'
  359.         stat = stat
  360.         import dos
  361.     elif 'os2' in names:
  362.         sep = '\\'
  363.         stat = stat
  364.         import os2
  365.     elif 'mac' in names:
  366.         stat = stat
  367.         import mac
  368.         
  369.         def join(a, b):
  370.             if a == '':
  371.                 return b
  372.             
  373.             if ':' not in a:
  374.                 a = ':' + a
  375.             
  376.             if a[-1:] != ':':
  377.                 a = a + ':'
  378.             
  379.             return a + b
  380.  
  381.     else:
  382.         raise ImportError, 'no os specific module found'
  383.     if join is None:
  384.         
  385.         def join(a, b, sep = sep):
  386.             if a == '':
  387.                 return b
  388.             
  389.             lastchar = a[-1:]
  390.             if lastchar == '/' or lastchar == sep:
  391.                 return a + b
  392.             
  393.             return a + sep + b
  394.  
  395.     
  396.     _os_stat = stat
  397.     _os_path_join = join
  398.  
  399.  
  400. def _os_path_isdir(pathname):
  401.     '''Local replacement for os.path.isdir().'''
  402.     
  403.     try:
  404.         s = _os_stat(pathname)
  405.     except OSError:
  406.         return None
  407.  
  408.     return s.st_mode & 61440 == 16384
  409.  
  410.  
  411. def _timestamp(pathname):
  412.     '''Return the file modification time as a Long.'''
  413.     
  414.     try:
  415.         s = _os_stat(pathname)
  416.     except OSError:
  417.         return None
  418.  
  419.     return long(s.st_mtime)
  420.  
  421.  
  422. class BuiltinImporter(Importer):
  423.     
  424.     def get_code(self, parent, modname, fqname):
  425.         if parent:
  426.             return None
  427.         
  428.         if imp.is_builtin(modname):
  429.             type = imp.C_BUILTIN
  430.         elif imp.is_frozen(modname):
  431.             type = imp.PY_FROZEN
  432.         else:
  433.             return None
  434.         module = imp.load_module(modname, None, modname, ('', '', type))
  435.         return (0, module, { })
  436.  
  437.  
  438.  
  439. class _FilesystemImporter(Importer):
  440.     
  441.     def __init__(self):
  442.         self.suffixes = []
  443.  
  444.     
  445.     def add_suffix(self, suffix, importFunc):
  446.         if not callable(importFunc):
  447.             raise AssertionError
  448.         self.suffixes.append((suffix, importFunc))
  449.  
  450.     
  451.     def import_from_dir(self, dir, fqname):
  452.         result = self._import_pathname(_os_path_join(dir, fqname), fqname)
  453.         if result:
  454.             return self._process_result(result, fqname)
  455.         
  456.  
  457.     
  458.     def get_code(self, parent, modname, fqname):
  459.         if not parent:
  460.             raise AssertionError
  461.         return self._import_pathname(_os_path_join(parent.__pkgdir__, modname), fqname)
  462.  
  463.     
  464.     def _import_pathname(self, pathname, fqname):
  465.         if _os_path_isdir(pathname):
  466.             result = self._import_pathname(_os_path_join(pathname, '__init__'), fqname)
  467.             if result:
  468.                 values = result[2]
  469.                 values['__pkgdir__'] = pathname
  470.                 values['__path__'] = [
  471.                     pathname]
  472.                 return (1, result[1], values)
  473.             
  474.             return None
  475.         
  476.         for suffix, importFunc in self.suffixes:
  477.             filename = pathname + suffix
  478.             
  479.             try:
  480.                 finfo = _os_stat(filename)
  481.             except OSError:
  482.                 continue
  483.  
  484.             return importFunc(filename, finfo, fqname)
  485.         
  486.  
  487.  
  488.  
  489. def py_suffix_importer(filename, finfo, fqname):
  490.     file = filename[:-3] + _suffix
  491.     t_py = long(finfo[8])
  492.     t_pyc = _timestamp(file)
  493.     code = None
  494.     if t_pyc is not None and t_pyc >= t_py:
  495.         f = open(file, 'rb')
  496.         if f.read(4) == imp.get_magic():
  497.             t = struct.unpack('<I', f.read(4))[0]
  498.             if t == t_py:
  499.                 code = marshal.load(f)
  500.             
  501.         
  502.         f.close()
  503.     
  504.     if code is None:
  505.         file = filename
  506.         code = _compile(file, t_py)
  507.     
  508.     return (0, code, {
  509.         '__file__': file })
  510.  
  511.  
  512. class DynLoadSuffixImporter:
  513.     
  514.     def __init__(self, desc):
  515.         self.desc = desc
  516.  
  517.     
  518.     def import_file(self, filename, finfo, fqname):
  519.         fp = open(filename, self.desc[1])
  520.         module = imp.load_module(fqname, fp, filename, self.desc)
  521.         module.__file__ = filename
  522.         return (0, module, { })
  523.  
  524.  
  525.  
  526. def _print_importers():
  527.     items = sys.modules.items()
  528.     items.sort()
  529.     for name, module in items:
  530.         if module:
  531.             print name, module.__dict__.get('__importer__', '-- no importer')
  532.             continue
  533.         print name, '-- non-existent module'
  534.     
  535.  
  536.  
  537. def _test_revamp():
  538.     ImportManager().install()
  539.     sys.path.insert(0, BuiltinImporter())
  540.  
  541.